package com.kehao.leetcode.jzoffer;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

public class JZSolution066 {

    private static class MapSum {

        private Map<String,Integer> map;

        public MapSum() {
            map = new HashMap<>();
        }

        public void insert(String key, int val) {
            map.put(key,val);
        }

        public int sum(String prefix) {
             return map.entrySet().stream().filter(en->en.getKey().startsWith(prefix)).map(en->en.getValue()).reduce((v1,v2)->v1+v2).orElse(0);
        }
    }

    @Test
    public void test01(){
        MapSum mapSum = new MapSum();
        mapSum.insert("a",3);
        mapSum.sum("ap");
    }
}
